package cn.wjssl.domain.strategy.model.entity;

import cn.wjssl.types.common.Constants;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 策略规则详情实体类
 */
@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class StrategyRuleEntity {

    /**
     * 抽奖策略ID
     */
    private Long strategyId;
    /**
     * 抽奖奖品ID【规则类型为策略，则不需要奖品ID】
     */
    private Integer awardId;
    /**
     * 抽象规则类型；1-策略规则、2-奖品规则
     */
    private Integer ruleType;
    /**
     * 抽奖规则类型【rule_random - 随机值计算、rule_lock - 抽奖几次后解锁、rule_luck_award - 幸运奖(兜底奖品)】
     */
    private String ruleModel;
    /**
     * 抽奖规则比值
     */
    private String ruleValue;
    /**
     * 抽奖规则描述
     */
    private String ruleDesc;

    /**
     * 获取权重值
     * 数据案例；4000:102,103,104,105 5000:102,103,104,105,106,107 6000:102,103,104,105,106,107,108,109
     * 最后返回的map 应该是 [{"4000:102,103,104,105":[102,103,104,105]}, ...]
     */
    public Map<String, List<Integer>> getRuleWeightValues() {
        // 如果该表中没有rule_weight, 直接返回null
        if (!ruleModel.equals("rule_weight")) return null;
        // 根据 空格, 分割权重信息
        String[] ruleValueGroups = ruleValue.split(Constants.SPACE);
        Map<String, List<Integer>> resultMap = new HashMap<>();
        for (String ruleValueGroup : ruleValueGroups) {
            // 检查输入是否为空
            if (ruleValueGroup == null || ruleValueGroup.isEmpty()) {
                return resultMap;
            }
            // 根据冒号分割 各自的信息
            String[] parts = ruleValueGroup.split(Constants.COLON);
            // 如果不是一一对应,也就是不是 xx:yy,yy2 的格式, 就报异常
            if (parts.length != 2) {
                throw new IllegalArgumentException("rule_weight rule_rule invalid input format" + ruleValueGroup);
            }
            // 解析对应的奖品列表
            String[] awardIdStrings = parts[1].split(Constants.SPLIT);
            // 因为awardId 是int类型,所以需要转换一下
            List<Integer> awardIds = Arrays.stream(awardIdStrings)
                    .map(Integer::parseInt)
                    .collect(Collectors.toList());
            // 将键和值放入Map中
            resultMap.put(ruleValueGroup, awardIds);
        }
        return resultMap;
    }
}
